#ifndef ML_SUBCLASS_FOLDER_H_
#define ML_SUBCLASS_FOLDER_H_

#include "ZMLAlgorithm.h"
#include <vector>
#include <set>

GLOBAL_NAMESPACE_BEGIN
NAMESPACE_MACHINE_LEARNING_BEGIN

class MLSubClassFolder
{
public:
    typedef std::vector<int> SubClass;
public:
    MLSubClassFolder();
    ~MLSubClassFolder();

    void addSubClass(const std::vector<int>& subClass, int newAddedClass);
    const SubClass* getSubClass(int idx) const;
    int  getSubClassSize() const;
    int  getTheNewAddedClass(int idx) const;

private:
    std::vector<SubClass> subClasses_;
    std::vector<int> newAddedClass_;
};

NAMESPACE_MACHINE_LEARNING_END
GLOBAL_NAMESPACE_END

#endif//ML_SUBCLASS_FOLDER_H_